python program sort words